cd ../..

gpus=$1

for ood_size in 0 24 48 96 192
do 
    python semi_main.py \
        --gpus ${gpus}, \
        --opt_conf ./configs/opt_joint.yaml \
        --exp_name semi/BYOLA_3C+R/w0.1_o${ood_size}_i24 \
        --feat_dim 256 \
        --freeze_cnn_layer '[0, 1, 2]' \
        --batch_sizes "[0, 12, 12, 24, ${ood_size}]" \
        --augmentation BYOLA \
        --self_target crnn 
        # --const_max 0.1

done   